from wordcloud import WordCloud
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

# 读取文本文件
with open("D:\\嘟嘟\\历史数据\\python学习\\实践项目1 jieba分词\\绘制词云图用文档\\唐诗\\唐诗三百首.txt", "r", encoding="utf-8") as file:
    text = file.read()

# 读取五角星蒙版图片并转换为numpy数组
star_mask = np.array(Image.open('star_mask.png'))

# 如果您有一个镰刀形状的蒙版图片，您应该这样加载它
# 假设您的镰刀蒙版图片名为 'sickle_mask.png'
sickle_mask = np.array(Image.open('sickle_mask.png'))

# 创建词云对象，设置参数
wordcloud = WordCloud(
    font_path="D:\\嘟嘟\\历史数据\\python学习\\msyh.ttc",  # 中文字体路径
    mask=sickle_mask,          # 镰刀蒙版
    background_color='blue' # 背景颜色
).generate(text)

# 显示词云图
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis('off')
plt.show()